package io.binghe.im.openai.domain.chapgpt.model.request;

import java.io.Serializable;

/**
 * @author binghe(微信 : hacker_binghe)
 * @version 1.0.0
 * @description 调用OpenAI接口的参数
 * @github https://github.com/binghe001
 * @copyright 公众号: 冰河技术
 */
public class AIRequest implements Serializable {

    private String model = "text-davinci-003";
    private String prompt;
    private Integer temperature = 0;
    private Integer max_tokens = 1024;

    public AIRequest() {
    }

    public AIRequest(String prompt) {
        this.prompt = prompt;
    }

    public void setPrompt(String prompt) {
        this.prompt = prompt;
    }

    public String getModel() {
        return model;
    }

    public String getPrompt() {
        return prompt;
    }

    public Integer getTemperature() {
        return temperature;
    }

    public Integer getMax_tokens() {
        return max_tokens;
    }
}
